import java.util.*;

/**
 * @author LKQ
 * @date 2022/6/27 17:00
 * @description 如果题目演变为 求和为k的子数组的最大长度，那么同样采取HashMap的方式
 */
public class Solution2 {
    public static void main(String[] args) {
        int[] arr = {1, -2 , 1, 1, 1, 1};
        Solution2 solution2 = new Solution2();
        solution2.maxlenEqualK(arr, 0);
    }
    /**
     * 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
     *
     * max length of the subarray sum = k
     * @param arr int整型一维数组 the array
     * @param k int整型 target
     * @return int整型
     */
    public int maxlenEqualK (int[] arr, int k) {
        long sum = 0;
        int len = Integer.MIN_VALUE;
        // 存储和为sum的数组长度
        Map<Long, Integer> map = new HashMap<>();
        // 处理边界
        map.put(0L, -1);

        for (int i = 0; i < arr.length; i++) {
            sum += arr[i];
            if (!map.containsKey(sum)) {
                // 如果不包含这个和，那么加入map，因为记录的是最长数组
                map.put(sum, i);
            }
            if (map.containsKey(sum - k) && i - map.get(sum - k) > len) {
                len = i - map.get(sum - k);
            }
        }
        return len;
    }
}
